package com.jnc.plug.mybatis.mp.generator.config;

import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.PackageConfig;

/**
 * @Description
 * @Author fanlei
 * @Version 1.0
 * @Date 2019/9/28
 */
public class CustomPackageConfig {
    private AutoGenerator mpg;

    public CustomPackageConfig(AutoGenerator mpg) {
        this.mpg = mpg;
    }

    public void invoke() {
        mpg.setPackageInfo(
                // 包配置
                new PackageConfig()
                        .setParent(Context.parent)// 自定义包路径
                        .setModuleName(Context.moduleName)
                        .setEntity(Context.entity)
                        .setController(Context.controller)// 这里是控制器包名，默认 web
                        .setService(Context.service)
                        .setServiceImpl(Context.serviceImpl)
                        .setMapper(Context.mapper)
                        .setXml(Context.xml)
        );
    }
}
